package datastructure.recursive;

/**
 * 8皇后问题
 *
 * @author jack.wu
 * @version 1.0
 * @date 2020-03-24
 */
public class Queen8 {

    private static int count = 0;

    /**
     * 最多有多少皇后
     */
    private int max = 8;

    /**
     * 存放皇后位置的结果
     * array[n] + 1 表示第几列
     * n+1表示第几行
     * n+1表示第几个皇后
     */
    private int[] array = new int[max];

    public static void main(String[] args) {
        Queen8 queen8 = new Queen8();
        queen8.check(0);
        System.out.println(count);
    }

    /**
     * 放皇后
     * 需要注意 check 是每一次递归，进入check中都有for (int i = 0 ;i<max;i++) 因此会回溯
     *
     * @param n 从哪开始
     */
    private void check(int n) {
        if (n == max) {
            // 8皇后已经放好
            print();
            return;
        }
        for (int i = 0; i < max; i++) {
            // 先把当前皇后放到该行的第0列
            array[n] = i;
            if (judge(n)) {
                // 不冲突 接着放n+1个皇后
                check(n + 1);
            }
            // 如果冲突，就继续执行 array[n] = i 向后一列移动
        }
    }

    /**
     * 判断是否冲突
     *
     * @param n 表示第几个皇后
     * @return 冲突 false 不冲突 true
     */
    private boolean judge(int n) {
        for (int i = 0; i < n; i++) {
            // array[i] == array[n] 表示列相等  上面定义好的
            // Math.abs(n - i) == Math.abs(array[n] - array[i]) 表示在一条斜线上  斜率 k = 1  行的差 = 列的差
            // 因为 n 表示行 所以不用判断行
            if (array[i] == array[n] || Math.abs(n - i) == Math.abs(array[n] - array[i])) {
                return false;
            }
        }
        return true;
    }

    /**
     * 打印结果
     */
    private void print() {
        count++;
        for (int i = 0; i < max; i++) {
            System.out.print(array[i] + " ");
        }
        System.out.println();
    }
}

